package servlet;

import pojo.User;
import service.UserService;

import javax.servlet.*;
import javax.servlet.http.*;
import javax.servlet.annotation.*;
import java.io.IOException;
//登录

@WebServlet(name = "LoginServlet", value = "/login")
public class LoginServlet extends HttpServlet {

    private UserService userService = new UserService();

    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String username = request.getParameter("username");
        String password = request.getParameter("password");
        //接收rememberMe
        String rememberMe = request.getParameter("rememberMe");
        User user = userService.login(username,password);
        if (user!=null){
            //完成前端的 记住我 功能
            if ("1".equals(rememberMe)){
                //向前端发送cookie记住用户名和密码的信息
                Cookie cookieUsername = new Cookie("username", username);
                Cookie cookiePassword = new Cookie("password", password);
                cookieUsername.setMaxAge(7*24*3600);
                cookiePassword.setMaxAge(7*24*3600);
                response.addCookie(cookieUsername);
                response.addCookie(cookiePassword);

            }else {
                //清除cookie已有的信息
                Cookie cookieUsername = new Cookie("username",null);
                Cookie cookiePassword = new Cookie("password",null);
                cookieUsername.setMaxAge(0);
                cookiePassword.setMaxAge(0);
                response.addCookie(cookieUsername);
                response.addCookie(cookiePassword);

            }

            //登录成功 1.存session 2.重定向到首页
            HttpSession session = request.getSession();
            session.setAttribute("LOGIN_USER",user);
            String contextPath = request.getContextPath();
            response.sendRedirect(contextPath+"/selectAll");
        }else {
            //登录失败 1，存失败消息 2.转发到登录页
            String msg = "用户名或者密码错误！";
            request.setAttribute("LOGIN_MSG",msg);
            request.getRequestDispatcher("/login.jsp").forward(request,response);
        }

    }
}
